/*
 * Copyright 2016-2017 TVI Go Easy.
 * Created on 2017/10/16 09:58
 */
package com.gitee.mechanic.mybatis.utils;

import com.baomidou.mybatisplus.plugins.Page;

import java.util.List;

/**
 * Mybatis plus 分页辅助工具类
 *
 * @author mechanic
 */
public class PageUtils {
    /**
     * 获取分页
     *
     * @param current 当前页
     * @param <T>     分页实体
     * @return 分页信息
     */
    public static <T> Page<T> getPage(int current) {
        Page<T> page = new Page<>();
        page.setCurrent(current);
        return page;
    }

    /**
     * 获取分页
     *
     * @param current 当前页
     * @param size    分页记录数
     * @param <T>     分页实体
     * @return 分页信息
     */
    public static <T> Page<T> getPage(int current, int size) {
        Page<T> page = new Page<>();
        page.setCurrent(current);
        page.setSize(size);
        return page;
    }

    /**
     * 获取分页
     *
     * @param current      当前页
     * @param size         分页记录数
     * @param orderByField 排序字段
     * @param <T>          分页实体
     * @return 分页信息
     */
    public static <T> Page<T> getPage(int current, int size, String orderByField) {
        Page<T> page = new Page<>();
        page.setCurrent(current);
        page.setSize(size);
        page.setOrderByField(orderByField);
        return page;
    }

    /**
     * 将entity分页转成 dto分页
     *
     * @param sourcePage ENTITY 分页信息
     * @param records    记录
     * @param <T>        DTO
     * @param <S>        ENTITY
     * @return DTO 分页信息
     */
    public static <T, S> Page<T> getPage(Page<S> sourcePage, List<T> records) {
        Page<T> page = new Page<>(sourcePage.getCurrent(), sourcePage.getSize(), sourcePage.getOrderByField());
        page.setTotal(sourcePage.getTotal());
        page.setCondition(sourcePage.getCondition());
        page.setRecords(records);
        page.getPages();
        return page;
    }
}
